# 插入数据
import random
from pymilvus import Collection, connections


# 连接到Milvus数据库 注意命名关键字时db_name！
connections.connect("default", host="localhost", port="19530", db_name="test")
# 获取名为example的集合对象
collection = Collection("example")

# 生成随机向量数据
# 创建10个向量，每个向量包含128个随机浮点数
vectors = [[random.random() for _ in range(128)] for _ in range(10)]

# 只插入向量字段（主键 auto_id=True可省略）
# 将向量数据组织成插入数据
data = [vectors]

# 插入数据
# 将数据插入到集合中，并获取插入结果
mr = collection.insert(data)

# 打印实际插入的向量数量
print("插入向量数量：", mr.insert_count)

# 向量数据落盘
collection.flush()

